package com.tan00xu.utils;

import io.jsonwebtoken.Jwt;
import io.jsonwebtoken.JwtBuilder;
import io.jsonwebtoken.Jwts;
import io.jsonwebtoken.SignatureAlgorithm;
import org.joda.time.DateTime;

import java.util.Date;
import java.util.HashMap;
import java.util.Map;

public class JWTUtils {

    /**
     * 秘钥
     */
    private static final String JWT_TOKEN = "TAN00XU@ym";

    /**
     * 创建token
     * @param userId 用户的id
     * @return 加密的token
     */
    public static String createToken(Long userId) {
        Map<String, Object> claims = new HashMap<>();
        claims.put("userId", userId);
        JwtBuilder jwtBuilder = Jwts.builder()
                .signWith(SignatureAlgorithm.HS256, JWT_TOKEN) // 签发算法，秘钥为JWT_TOKEN
                .setClaims(claims) // body数据，要唯一
                .setIssuedAt(new Date()) // 设置签发时间
                .setExpiration(new Date(System.currentTimeMillis() + 24 * 60 * 60 * 1000));// 一天的有效时间
        //得到token并返回
        return jwtBuilder.compact();
    }

    /**
     * 检查token是否合法
     * @param token 前端传来的token
     * @return Map<String, Object>
     */
    public static Map<String, Object> checkToken(String token) {
        try {
            Jwt parse = Jwts.parser().setSigningKey(JWT_TOKEN).parse(token);
            return (Map<String, Object>) parse.getBody();
        } catch (Exception e) {
            CmdOutputInformationUtils.error("token验证失败，时间："+new DateTime(System.currentTimeMillis()));
            e.printStackTrace();
        }
        return null;

    }

    public static void main(String[] args) {
        String token = JWTUtils.createToken(100L);
        System.out.println(token);
        Map<String, Object> stringObjectMap = JWTUtils.checkToken(token);
        System.out.println(stringObjectMap+"=========");
//        assert stringObjectMap != null;
        System.out.println(stringObjectMap.get("userId"));

    }

}
